package com.celesea.framework.util;

import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.HanyuPinyinVCharType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;

/**
 * @author JackYuan
 * @create 2017-11-04 10:01
 */
public class PinyinKit {

    public static final String NOT_PY_CHAR = "#";
    public static final char LOWER_CHAR_A = 'a';
    public static final char LOWER_CHAR_Z = 'z';
    public static final char UPPER_CHAR_A = 'A';
    public static final char UPPER_CHAR_Z = 'Z';

    public static String getPingYin(String src) {
        char[] t1 = null;
        t1 = src.toCharArray();
        String[] t2 = new String[t1.length];
        HanyuPinyinOutputFormat t3 = new HanyuPinyinOutputFormat();
        t3.setCaseType(HanyuPinyinCaseType.LOWERCASE);
        t3.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
        t3.setVCharType(HanyuPinyinVCharType.WITH_V);
        String t4 = "";
        int t0 = t1.length;
        try {
            for (int i = 0; i < t0; i++) {
                // 判断是否为汉字字符
                if (Character.toString(t1[i]).matches("[\\u4E00-\\u9FA5]+")) {
                    t2 = PinyinHelper.toHanyuPinyinStringArray(t1[i], t3);
                    t4 += t2[0];
                } else {
                    t4 += Character.toString(t1[i]);
                }
            }
            return t4;
        } catch (BadHanyuPinyinOutputFormatCombination e1) {
            e1.printStackTrace();
        }
        return t4;
    }

    public static String getPinYinHeadChar(String str) {
        String convert = "";
        for (int j = 0; j < str.length(); j++) {
            char word = str.charAt(j);
            String[] pinyinArray = PinyinHelper.toHanyuPinyinStringArray(word);
            if (pinyinArray != null) {
                convert += pinyinArray[0].charAt(0);
            } else {
                convert += word;
            }
        }
        return convert;
    }

    public static String getPinyinFirst(String str) {
        String head = getPinYinHeadChar(str);
        if (StringKit.isNotEmpty(head)) {
            char c = head.charAt(0);
            boolean isEnglisthChar = c >= LOWER_CHAR_A && c <= LOWER_CHAR_Z || c >= UPPER_CHAR_A && c <= UPPER_CHAR_Z;
            return isEnglisthChar ? String.valueOf(c).toUpperCase() : NOT_PY_CHAR;
        }
        return null;
    }

}
